package com.suntoon.framework.file;

import cn.hutool.core.io.FileUtil;
import cn.hutool.core.io.file.FileNameUtil;
import com.suntoon.framework.exception.MessageCenterException;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;

import javax.activation.MimetypesFileTypeMap;
import javax.servlet.http.HttpServletResponse;
import java.io.*;

/**
 * @Author: LEAVES
 * @Version 1.0
 * @Date: 2021年06月25日 14时38分10秒
 * @Description:    文件下载
 */
@Slf4j
@Component
public class FileDownloadUtil {

    /**
     * 文件下载
     *
     * @param path     文件路径
     * @param response 返回消息
     * @return
     */
    public void downFile(String path, HttpServletResponse response) {
        log.info("文件下载：" + path);
        //获取文件名    引用hutool工具类库
        String filename = FileNameUtil.getName(FileUtil.file(path));
        log.info("文件名称：" + filename);
        File file = new File(path);
        if (file.exists()) {
            byte[] buffer = new byte[1024];
            FileInputStream fis = null;
            BufferedInputStream bis = null;
            try {
                // 设置强制下载不打开
                response.setContentType("application/force-download;charset=UTF-8");
                // 设置文件名
                response.addHeader("Content-Disposition", "attachment;fileName=" + filename);
                // 设置信息给客户端不解析
                String type = new MimetypesFileTypeMap().getContentType(filename);
//              设置contenttype，即告诉客户端所发送的数据属于什么类型
                response.setHeader("Content-type", type);
//              设置编码
                String hehe = new String(filename.getBytes("utf-8"), "iso-8859-1");
//              设置扩展头，当Content-Type 的类型为要下载的类型时 , 这个信息头会告诉浏览器这个文件的名字和类型。
                response.setHeader("Content-Disposition", "attachment;filename=" + hehe);
                fis = new FileInputStream(file);
                bis = new BufferedInputStream(fis);
                OutputStream os = response.getOutputStream();
                int i = bis.read(buffer);
                while (i != -1) {
                    os.write(buffer, 0, i);
                    i = bis.read(buffer);
                }
            } catch (FileNotFoundException fe) {
                fe.printStackTrace();
                log.error("### 下载的文件不存在 ###");
                throw new MessageCenterException("下载的文件不存在", fe);
            } catch (IOException ie) {
                ie.printStackTrace();
                log.error("### 文件下载出错啦 ###");
                throw new MessageCenterException("下载的文件不存在", ie);
            } finally {
                if (bis != null) {
                    try {
                        bis.close();
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }
                if (fis != null) {
                    try {
                        fis.close();
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }
            }
        } else {
            log.error("### 文件下载出错啦 ###");
            throw new MessageCenterException("文件下载出错啦");
        }
    }
}
